package com.dl.utils;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * @author DL
 * @create 2021-04-11 11:35
 * 判断文件类型工具类
 */
public class TypeUtils {

    /**
     * 传入自定义数组判断类型
     */
    public static boolean judgeVideo(String[] typeArr,String type){
        List<String> types = Arrays.asList(typeArr);
        return types.contains(type);
    }

    /**
     * 判断是否为视频类型
     */
    public static boolean judgeVideo(String type){
        List<String> fileType = new ArrayList<>();
        fileType.add("video/mp4");//mp4
        fileType.add("video/x-msvideo");//avi
        fileType.add("video/mpeg");//mpeg
        fileType.add("video/x-flv");//flv
        fileType.add("video/x-matroska");//mkv
        fileType.add("video/quicktime");//mov
        return fileType.contains(type);
    }

    /**
     * 判断是否为图像类型
     */
    public static boolean judgeImage(String type){
        List<String> fileType = new ArrayList<>();
        fileType.add("image/jpg");
        fileType.add("image/jpeg");
        fileType.add("image/png");
        fileType.add("image/bmp");
        return fileType.contains(type);
    }
}
